package com.css.zfzx.fxpg.modules.vulnerability.danger.controller;

import com.alibaba.fastjson.JSONObject;
import com.css.bpm.platform.base.response.RestResponse;
import com.css.zfzx.fxpg.modules.preestimate.workspace.viewobjects.FxPreestimateWorkspaceQueryParams;
import com.css.zfzx.fxpg.modules.vulnerability.danger.repository.entity.FxVulnerabilityDangerEntity;
import com.css.zfzx.fxpg.modules.vulnerability.danger.service.FxVulnerabilityDangerService;
import com.css.zfzx.fxpg.modules.vulnerability.danger.viewobjects.FxVulnerabilityDangerQueryParams;
import com.css.bpm.platform.utils.PlatformPageUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;

/**
 * @author andy
 * @date 2020-12-10
 */
@Slf4j
@RestController
@RequestMapping("/fxVulnerabilityDangers")
public class FxVulnerabilityDangerController {
    @Autowired
    private FxVulnerabilityDangerService fxVulnerabilityDangerService;

    @GetMapping("/queryFxVulnerabilityDangers")
    public RestResponse queryFxVulnerabilityDangers(HttpServletRequest request, FxVulnerabilityDangerQueryParams queryParams) {
        RestResponse response = null;
        try{
            int curPage = Integer.valueOf(request.getParameter(PlatformPageUtils.CUR_PAGE));
            int pageSize = Integer.valueOf(request.getParameter(PlatformPageUtils.PAGE_SIZE));
            JSONObject jsonObject = fxVulnerabilityDangerService.queryFxVulnerabilityDangers(queryParams,curPage,pageSize);
            response = RestResponse.succeed(jsonObject);
        }catch (Exception e){
            String errorMessage = "查询失败!";
            log.error(errorMessage,e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }

    @GetMapping("{id}")
    public RestResponse getFxVulnerabilityDanger(@PathVariable String id) {
        RestResponse response = null;
        try{
            FxVulnerabilityDangerEntity fxVulnerabilityDanger = fxVulnerabilityDangerService.getFxVulnerabilityDanger(id);
            response = RestResponse.succeed(fxVulnerabilityDanger);
        }catch (Exception e){
            String errorMessage = "获取失败!";
            log.error(errorMessage,e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }
    @PostMapping
    public RestResponse saveFxVulnerabilityDanger(@RequestBody FxVulnerabilityDangerEntity fxVulnerabilityDanger) {
        RestResponse response = null;
        JSONObject json = new JSONObject();
        try{
            fxVulnerabilityDangerService.saveFxVulnerabilityDanger(fxVulnerabilityDanger);
            json.put("message", "新增成功!");
            response = RestResponse.succeed(json);
        }catch (Exception e){
            String errorMessage = "新增失败!";
            log.error(errorMessage,e);
            response = RestResponse.fail(errorMessage);
        }
        return response;

    }
    @PutMapping
    public RestResponse updateFxVulnerabilityDanger(@RequestBody FxVulnerabilityDangerEntity fxVulnerabilityDanger)  {
        RestResponse response = null;
        JSONObject json = new JSONObject();
        try{
            fxVulnerabilityDangerService.updateFxVulnerabilityDanger(fxVulnerabilityDanger);
            json.put("message", "修改成功!");
            response = RestResponse.succeed(json);
        }catch (Exception e){
            String errorMessage = "修改失败!";
            log.error(errorMessage,e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }
    @DeleteMapping
    public RestResponse deleteFxVulnerabilityDangers(@RequestParam List<String> ids) {
        RestResponse response = null;
        JSONObject json = new JSONObject();
        try{
            fxVulnerabilityDangerService.deleteFxVulnerabilityDangers(ids);
            json.put("message", "删除成功!");
            response = RestResponse.succeed(json);
        }catch (Exception e){
            String errorMessage = "删除失败!";
            log.error(errorMessage,e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }

    @GetMapping("/getValidDictItemsByDictCode/{dictCode}")
    public RestResponse getValidDictItemsByDictCode(@PathVariable String dictCode) {
        RestResponse restResponse = null;
        try {
            restResponse = RestResponse.succeed(fxVulnerabilityDangerService.getValidDictItemsByDictCode(dictCode));
        } catch (Exception e) {
            String errorMsg = "字典项获取失败!";
            log.error(errorMsg, e);
            restResponse = RestResponse.fail(errorMsg);
        }
        return restResponse;
    }

    @GetMapping("/getFxVulnerabilityDangers")
    public RestResponse getFxVulnerabilityDangers(FxVulnerabilityDangerQueryParams queryParams) {
        RestResponse response = null;
        try{
            response = RestResponse.succeed(fxVulnerabilityDangerService.getFxVulnerabilityDangers(queryParams));
        }catch (Exception e){
            String errorMessage = "查询失败!";
            log.error(errorMessage,e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }

    /**
     * 导入
     * @param file
     * @return
     */
    @PostMapping("/import")
    public RestResponse importExcel(@RequestParam("file") MultipartFile file) {
        RestResponse restResponse = null;
        try {
            String s = fxVulnerabilityDangerService.importExcel(file);
            restResponse = RestResponse.succeed(s);
        } catch (Exception e) {
            String errorMsg = "导入失败!";
            log.error(errorMsg, e);
            restResponse = RestResponse.fail(errorMsg);
        }
        return restResponse;
    }

    /**
     * 导出
     *
     * @param queryParams
     * @param response
     * @return
     */
    @GetMapping("/export")
    public RestResponse exportExcel(FxVulnerabilityDangerQueryParams queryParams, HttpServletResponse response) {
        RestResponse restResponse = null;
        try {
            String s = fxVulnerabilityDangerService.exportExcel(queryParams, response);
            restResponse = RestResponse.succeed(s);
        } catch (Exception e) {
            String errorMsg = "导出失败!";
            log.error(errorMsg, e);
            restResponse = RestResponse.fail(errorMsg);
        }
        return restResponse;
    }

}